<?php
/**
 * Desc: php操作mysql的封装类
 * Author wzb
 * Date: 2018/08/29
 * 连接模式：PDO
 */
class DBPDO2 {
    protected static $_dbh = null; //静态属性,所有数据库实例共用,避免重复连接数据库
    protected $_dbType = 'mysql';
    protected $_pconnect = true; //是否使用长连接
    protected $_host = 'localhost';
    protected $_port = 3306;
    protected $_user = 'root';
    protected $_pass = 'root';
    protected $_dbName = null; //数据库名
    protected $_sql = false; //最后一条sql语句
    protected $_where = '';
    protected $_order = '';
    protected $_limit = '';
    protected $_field = '*';
    protected $_clear = 0; //状态，0表示查询条件干净，1表示查询条件污染
    protected $_trans = 0; //事务指令数

    /**
     * 初始化类
     * @param array $conf 数据库配置
     */
    public function __construct() {
        class_exists('PDO') or die("PDO: class not exists.");
        $this->_host = 'rm-vy15f520u56n44eew985.mysql.zhangbei.rds.aliyuncs.com';
        $this->_port = '3306';
        $this->_user = 'ry_aliexpress';
        $this->_pass = 'Rymht321123!@';
        $this->_dbName = 'dulizhan';
        //连接数据库
        if ( is_null(self::$_dbh) ) {
            $this->_connect();
        }
    }

    /**
     * 连接数据库的方法
     */
    protected function _connect() {
        $dsn = $this->_dbType.':host='.$this->_host.';port='.$this->_port.';dbname='.$this->_dbName;
        $options = $this->_pconnect ? array(PDO::ATTR_PERSISTENT=>true) : array();
        try {
            $dbh = new PDO($dsn, $this->_user, $this->_pass, $options);
            $dbh->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);  //设置如果sql语句执行错误则抛出异常，事务会自动回滚
            //$dbh->setAttribute(PDO::ATTR_EMULATE_PREPARES, false); //禁用prepared statements的仿真效果(防SQL注入)
            $dbh->query("set character set 'UTF8'");
        } catch (PDOException $e) {
            die('Connection failed: ' . $e->getMessage());
        }
        //$dbh->exec('SET NAMES utf8');

        self::$_dbh = $dbh;
    }

    /**
    * 字段和表名添加 `符号
    * 保证指令中使用关键字不出错 针对mysql
    * @param string $value
    * @return string
    */
    protected function _addChar($value) {
        if ('*'==$value || false!==strpos($value,'(') || false!==strpos($value,'.') || false!==strpos($value,'`')) {
            //如果包含* 或者 使用了sql方法 则不作处理
        } elseif (false === strpos($value,'`') ) {
            $value = '`'.trim($value).'`';
        }
        return $value;
    }

    /**
    * 取得数据表的字段信息
    * @param string $tbName 表名
    * @return array
    */
    protected function _tbFields($tbName) {
        $sql = 'SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME="'.$tbName.'" AND TABLE_SCHEMA="'.$this->_dbName.'"';
        $stmt = self::$_dbh->prepare($sql);
        $stmt->execute();
        $result = $stmt->fetchAll(PDO::FETCH_ASSOC);
        $ret = array();
        foreach ($result as $key=>$value) {
            $ret[$value['COLUMN_NAME']] = 1;
        }
        return $ret;
    }

    /**
    * 过滤并格式化数据表字段
    * @param string $tbName 数据表名
    * @param array $data POST提交数据
    * @return array $newdata
    */
    protected function _dataFormat($tbName,$data) {
        if (!is_array($data)) return array();
        $table_column = $this->_tbFields($tbName);
        $ret=array();
        foreach ($data as $key=>$val) {
            if (!is_scalar($val)) continue; //值不是标量则跳过
            if (array_key_exists($key,$table_column)) {
                $key = $this->_addChar($key);
                if (is_int($val)) {
                    $val = intval($val);
                } elseif (is_float($val)) {
                    $val = floatval($val);
                } elseif (preg_match('/^\(\w*(\+|\-|\*|\/)?\w*\)$/i', $val)) {
                    // 支持在字段的值里面直接使用其它字段 ,例如 (score+1) (name) 必须包含括号
                    $val = $val;
                } elseif (is_string($val)) {
                    $val = '"'.addslashes($val).'"';
                }
                $ret[$key] = $val;
            }
        }
        return $ret;
    }

    /**
    * 执行查询 主要针对 SELECT, SHOW 等指令
    * @param string $sql sql指令
    * @return mixed
    */
    public function _doQuery($sql='',$type='all') {
        $this->_sql = $sql;
        $pdostmt = self::$_dbh->prepare($this->_sql); //prepare或者query 返回一个PDOStatement
        $pdostmt->execute();
        if($type == 'all'){
            $result = $pdostmt->fetchAll(PDO::FETCH_ASSOC);
        }else{
            $result = $pdostmt->fetch(PDO::FETCH_ASSOC);
        }

        return $result;
    }

    /**
    * 执行语句 针对 INSERT, UPDATE 以及DELETE,exec结果返回受影响的行数
    * @param string $sql sql指令
    * @return integer
    */
    protected function _doExec($sql) {
        $this->_sql = $sql;
        return self::$_dbh->exec($this->_sql);
    }

    /**
    * 执行sql语句，自动判断进行查询或者执行操作
    * @param string $sql SQL指令
    * @return mixed
    */
    public function doSql($sql='',$type='all') {
        $queryIps = 'INSERT|UPDATE|DELETE|REPLACE|CREATE|DROP|LOAD DATA|SELECT .* INTO|COPY|ALTER|GRANT|REVOKE|LOCK|UNLOCK';
        if (preg_match('/^\s*"?(' . $queryIps . ')\s+/i', $sql)) {
            echo date('Y-m-d H:i:s')." *【doExec】**$sql***\n";
            return $this->_doExec($sql);
        }
        else {
            //查询操作
            echo date('Y-m-d H:i:s')." *【doQuery】**$sql***\n";
            return $this->_doQuery($sql,$type);
        }
    }

    /**
    * 获取最近一次查询的sql语句
    * @return String 执行的SQL
    */
    public function getLastSql() {
        return $this->_sql;
    }

    /**
     * 插入方法
     * @param string $tbName 操作的数据表名
     * @param array $data 字段-值的一维数组
     * @return int 受影响的行数
     */
    public function insert($tbName,array $data){
        $data = $this->_dataFormat($tbName,$data);
        if (!$data) return;
        $sql = "insert into ".$tbName."(".implode(',',array_keys($data)).") values(".implode(',',array_values($data)).")";
        echo date('Y-m-d H:i:s')." *【insert】**$sql***\n";
        return $this->_doExec($sql);
    }

    /**
     * 删除方法
     * @param string $tbName 操作的数据表名
     * @return int 受影响的行数
     */
    public function delete($tbName) {
        //安全考虑,阻止全表删除
        if (!trim($this->_where)) return false;
        $sql = "delete from ".$tbName." ".$this->_where;
        echo date('Y-m-d H:i:s')." *【delete】**$sql***\n";
        $this->_clear = 1;
        $this->_clear();
        return $this->_doExec($sql);
    }

   //插入存在则更新
    public function Getinsert($arrData,$table) {
    $name = $values = '';
    $flag = $flagV = 1;
    $bindparams = $values2 = '';
    $true = is_array(current($arrData));//判断是否一次插入多条数据
    if($true) {
        //构建插入多条数据的sql语句
        foreach($arrData as $arr) {
            $values .= $flag ? '(' : ',(';
            $values2 .= $flag ? '(' : ',(';
            foreach($arr as $key => $value) {
                if($flagV) {
                    if($flag) $name .= "$key";
                    $values .= "'$value'";
                    $values2 .= "':$key'";
                    $flagV = 0;
                } else {
                    if($flag) $name .= ",$key";
                    $values .= ",'$value'";
                    $values2 .= ",':$key'";
                }
            }
            $values .= ') ';
            $values2 .= ') ';
            $flag = 0;
            $flagV = 1;
        }
    } else {
        //构建插入单条数据的sql语句
        foreach($arrData as $key => $value) {
            if($flagV) {
                $name = "$key";
                $values = "('$value'";
                $values2 = "(':$key'";
                $flagV = 0;
            } else {
                $name .= ",$key";
                $values .= ",'$value'";
                $values2 .= ",':$key'";
            }
        }
        $values .= ") ";
        $values2 .= ") ";
       $flagV2 = 1;
         foreach($arrData as $key => $value) {
            if($flagV2) {
                $name = "$key";
                $values_du = "$key='$value'";
                $values_du2 = "$key=':$key'";
                $flagV2 = 0;
            } else {
                $name .= ",$key";
                $values_du .= ",$key='$value'";
                $values_du2 .= ",$key=':$key'";
            }
        }
    }
    $strSql = "insert into $table ($name) values $values2 ON DUPLICATE KEY UPDATE $values_du2";
    echo date('Y-m-d H:i:s')." *【gensert】**$strSql***\n";
    return $this->_doQuery($strSql);
}

public function checkIsIsert($arrData,$table,$field,$filedvalue){
    $filedvalue=addslashes($filedvalue);
    $row = $this->where("$field='$filedvalue'")->select($table);
    if(count($row) > 0){
        //存在 更新
        echo $filedvalue."**存在***\n";
        return $this->where(array("$field"=>$filedvalue))->update($table,$arrData);
    }elseif (count($row) == 0) {
        //不存在 插入
        echo $filedvalue."***不存在***\n";
        return $this->insert($table,$arrData);
    }else{
        return false;
    }
}

    /**
     * 更新函数
     * @param string $tbName 操作的数据表名
     * @param array $data 参数数组
     * @return int 受影响的行数
     */
    public function update($tbName,array $data) {
        //安全考虑,阻止全表更新
        if (!trim($this->_where)) return false;
        $data = $this->_dataFormat($tbName,$data);
        if (!$data) return;
        $valArr = '';
        foreach($data as $k=>$v){
            $valArr[] = $k.'='.$v;
        }
        $valStr = implode(',', $valArr);
        $sql = "update ".trim($tbName)." set ".trim($valStr)." ".trim($this->_where);
        echo date('Y-m-d H:i:s')." *【update】**$sql***\n";
        return $this->_doExec($sql);
    }

    /**
     * 查询函数
     * @param string $tbName 操作的数据表名
     * @return array 结果集
     */
    public function select($tbName='',$type='all') {
        $sql = "select ".trim($this->_field)." from ".$tbName." ".trim($this->_where)." ".trim($this->_order)." ".trim($this->_limit);
        echo date('Y-m-d H:i:s')." *【select】**$sql***\n";
        $this->_clear = 1;
        $this->_clear();
        return $this->_doQuery(trim($sql),$type);
    }

    /**
     * @param mixed $option 组合条件的二维数组，例：$option['field1'] = array(1,'=>','or')
     * @return $this
     */
    public function where($option) {
        if ($this->_clear>0) $this->_clear();
        $this->_where = ' where ';
        $logic = 'and';
        if (is_string($option)) {
            $this->_where .= $option;
        }
        elseif (is_array($option)) {
            foreach($option as $k=>$v) {
                if (is_array($v)) {
                    $relative = isset($v[1]) ? "$v[1]" : '=';
                    $logic    = isset($v[2]) ? "$v[2]" : 'and';
                    $condition = ' ('.$this->_addChar($k).' '.$relative.' "'.$v[0].'") ';
                }
                else {
                    $logic = 'and';
                    $condition = ' ('.$this->_addChar($k).'="'.$v.'") ';
                }
                $this->_where .= isset($mark) ? $logic.$condition : $condition;
                $mark = 1;
            }
        }
        return $this;
    }

    /**
     * 设置排序
     * @param mixed $option 排序条件数组 例:array('sort'=>'desc')
     * @return $this
     */
    public function order($option) {
        if ($this->_clear>0) $this->_clear();
        $this->_order = ' order by ';
        if (is_string($option)) {
            $this->_order .= $option;
        }
        elseif (is_array($option)) {
            foreach($option as $k=>$v){
                $order = $this->_addChar($k).' '.$v;
                $this->_order .= isset($mark) ? ','.$order : $order;
                $mark = 1;
            }
        }
        return $this;
    }

    /**
     * 设置查询行数及页数
     * @param int $page pageSize不为空时为页数，否则为行数
     * @param int $pageSize 为空则函数设定取出行数，不为空则设定取出行数及页数
     * @return $this
     */
    public function limit($page,$pageSize=null) {
        if ($this->_clear>0) $this->_clear();
        if ($pageSize===null) {
            $this->_limit = "limit ".$page;
        }
        else {
            $pageval = intval( ($page - 1) * $pageSize);
            $this->_limit = "limit ".$pageval.",".$pageSize;
        }
        return $this;
    }

    /**
     * 设置查询字段
     * @param mixed $field 字段数组
     * @return $this
     */
    public function field($field){
        if ($this->_clear>0) $this->_clear();
        if (is_string($field)) {
            $field = explode(',', $field);
        }
        $nField = array_map(array($this,'_addChar'), $field);
        $this->_field = implode(',', $nField);
        return $this;
    }

    /**
     * 清理标记函数
     */
    protected function _clear() {
        $this->_where = '';
        $this->_order = '';
        $this->_limit = '';
        $this->_field = '*';
        $this->_clear = 0;
    }

    /**
     * 手动清理标记
     * @return $this
     */
    public function clearKey() {
        $this->_clear();
        return $this;
    }

    /**
    * 启动事务
    * @return void
    */
    public function startTrans() {
        //数据rollback 支持
        if ($this->_trans==0) self::$_dbh->beginTransaction();
        $this->_trans++;
        return;
    }

    /**
    * 用于非自动提交状态下面的查询提交
    * @return boolen
    */
    public function commit() {
        $result = true;
        if ($this->_trans>0) {
            $result = self::$_dbh->commit();
            $this->_trans = 0;
        }
        return $result;
    }

    /**
    * 事务回滚
    * @return boolen
    */
    public function rollback() {
        $result = true;
        if ($this->_trans>0) {
            $result = self::$_dbh->rollback();
            $this->_trans = 0;
        }
        return $result;
    }

    /**
    * 关闭连接
    * PHP 在脚本结束时会自动关闭连接。
    */
    public function close() {
        if (!is_null(self::$_dbh)) self::$_dbh = null;
    }

}
